RandomNormal

生成服从指定均值和标准差的正态分布随机数序列。

\[output_i \sim \mathcal{N}(\text{mean}, \text{scale}^2)\]
输入：

  • length - 输出数据长度。

  • mean - 正态分布的均值。

  • scale - 正态分布的标准差。

  • seed - 随机数种子，用于控制生成序列的随机性。

  • core_mask(int, 可选) - 核掩码（仅适用于共享存储版本）。

输出：

  • output - 生成的随机数结果地址。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持 fp32

  • MT7004 支持 fp32, fp16

  • 输出服从 \(\mathcal{N}(\text{mean}, \text{scale}^2)\) 分布。

  • 相同的随机种子会生成确定性一致的输出序列。

共享存储版本:

void fp_random_normal_s(float *output, int length, float mean, float scale, unsigned int seed, int core_mask)
void hp_random_normal_s(half *output, int length, float mean, float scale, unsigned int seed, int core_mask)

C 调用示例：

 1// FT78NE 示例
 2#include <stdio.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *output = (float *)0xA0000000;   // output 在 DDR 空间
 6    int length = 1024;
 7    float mean = 0.0f;
 8    float scale = 1.0f;
 9    unsigned int seed = 1234;
10    int core_mask = 0xff;
11    fp_random_normal_s(output, length, mean, scale, seed, core_mask);
12    return 0;
13}

私有存储版本:

void fp_random_normal_p(float *output, int length, float mean, float scale, unsigned int seed)
void hp_random_normal_p(half *output, int length, float mean, float scale, unsigned int seed)

C 调用示例：

 1// MT7004 示例
 2#include <stdio.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *output = (float *)0x10010000;
 6    int length = 1024;
 7    float mean = 0.0f;
 8    float scale = 1.0f;
 9    unsigned int seed = 1234;
10    fp_random_normal_p(output, length, mean, scale, seed);
11    return 0;
12}